# SPDX-License-Identifier: GPL-2.0
# Copyright (C) 2023-2024 Arm Technology (China) Co. Ltd.

MODULE_NAME := aipu
SRC_DIR := .
EXTRA_CFLAGS += -DKMD_VERSION=\"$(COMPASS_DRV_BTENVAR_KMD_VERSION)\" \
                   -I$(PWD)/armchina-npu/ -I$(PWD)/armchina-npu/include -I$(PWD)/armchina-npu/zhouyi \
                   -D$(BUILD_AIPU_VERSION_KMD) #-g

COMM_OBJ := $(SRC_DIR)/armchina-npu/aipu.o \
            $(SRC_DIR)/armchina-npu/aipu_common.o \
            $(SRC_DIR)/armchina-npu/aipu_io.o \
            $(SRC_DIR)/armchina-npu/aipu_irq.o \
            $(SRC_DIR)/armchina-npu/aipu_job_manager.o \
            $(SRC_DIR)/armchina-npu/aipu_mm.o \
            $(SRC_DIR)/armchina-npu/aipu_dma_buf.o \
            $(SRC_DIR)/armchina-npu/aipu_priv.o \
            $(SRC_DIR)/armchina-npu/aipu_tcb.o \
            $(SRC_DIR)/armchina-npu/zhouyi/zhouyi.o

ifeq ($(BUILD_AIPU_VERSION_KMD), BUILD_ZHOUYI_V1)
    AIPU_OBJ := $(SRC_DIR)/armchina-npu/zhouyi/v1.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v1v2_priv.o
    EXTRA_CFLAGS += -DCONFIG_ARMCHINA_NPU_ARCH_V1
else ifeq ($(BUILD_AIPU_VERSION_KMD), BUILD_ZHOUYI_V2)
    AIPU_OBJ := $(SRC_DIR)/armchina-npu/zhouyi/v2.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v1v2_priv.o
    EXTRA_CFLAGS += -DCONFIG_ARMCHINA_NPU_ARCH_V2
else ifeq ($(BUILD_AIPU_VERSION_KMD), BUILD_ZHOUYI_V3)
    AIPU_OBJ := $(SRC_DIR)/armchina-npu/zhouyi/v3.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v3_priv.o
    EXTRA_CFLAGS += -DCONFIG_ARMCHINA_NPU_ARCH_V3
else ifeq ($(BUILD_AIPU_VERSION_KMD), BUILD_ZHOUYI_V4)
    AIPU_OBJ := $(SRC_DIR)/armchina-npu/zhouyi/v4.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v4_priv.o
    EXTRA_CFLAGS += -DCONFIG_ARMCHINA_NPU_ARCH_V4
else
    #build all
    AIPU_OBJ := $(SRC_DIR)/armchina-npu/zhouyi/v1.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v2.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v3.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v4.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v1v2_priv.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v3_priv.o \
                $(SRC_DIR)/armchina-npu/zhouyi/v4_priv.o
    EXTRA_CFLAGS += -DCONFIG_ARMCHINA_NPU_ARCH_V1 \
                    -DCONFIG_ARMCHINA_NPU_ARCH_V2 \
                    -DCONFIG_ARMCHINA_NPU_ARCH_V3 \
                    -DCONFIG_ARMCHINA_NPU_ARCH_V4
endif

ifeq ($(BUILD_TARGET_PLATFORM_KMD), BUILD_PLATFORM_R329)
    INIT_OBJ  := $(SRC_DIR)/armchina-npu/r329/r329.o
else
    INIT_OBJ := $(SRC_DIR)/armchina-npu/default/default.o
endif

OBJS     := $(COMM_OBJ) $(AIPU_OBJ) $(SOC_OBJ)

ifneq ($(KERNELRELEASE),)
	obj-m := $(MODULE_NAME).o
	$(MODULE_NAME)-objs := $(INIT_OBJ) $(OBJS)
else
	COMPASS_DRV_BTENVAR_KPATH ?= /lib/modules/`uname -r`/build
	PWD :=$(shell pwd)

all:
	$(MAKE) -C $(COMPASS_DRV_BTENVAR_KPATH) M=$(PWD) modules
clean:
	$(MAKE) -C $(COMPASS_DRV_BTENVAR_KPATH) M=$(PWD) clean
endif
